from fit_py_ask_number_demo.configs.prompts import question_route_prompt
from fit_py_ask_number_demo.client import llm_client


class QuestionRoute:
    def __init__(self, query):
        self.query = query
        self.llm_func = llm_client.deepseek_client

    def run(self):
        user_content = f"<Question>\n{self.query}"
        resp = self.llm_func(question_route_prompt, user_content)
        return resp


def question_route_impl(query):
    question_router = QuestionRoute(query)
    return question_router.run()


if __name__ == '__main__':
    query = "24年北京收入"
    res = QuestionRoute(query).run()
    print(res)
